package com.sbtr.business.trainingmodel.mapper;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.Constants;
import com.sbtr.business.trainingmodel.entity.TrainingModelData;
import com.sbtr.open.trainingmodel.dto.ModelCountDto;
import com.sbtr.open.trainingmodel.dto.TrainingModelDataDto;
import com.sbtr.overview.dto.AllCountDto;
import com.sbtr.overview.dto.ModelTypeCountDto;
import org.apache.ibatis.annotations.Param;

import java.util.List;

/**
 * <p>
 * 训练模型表 Mapper 接口
 * </p>
 *
 * @author yuyh2

 */
public interface TrainingModelDataMapper extends BaseMapper<TrainingModelData> {
	IPage<TrainingModelDataDto> customPage(IPage<TrainingModelDataDto> page, @Param(Constants.WRAPPER) QueryWrapper<TrainingModelDataDto> wrapper);

	AllCountDto modelCount(String creater);

	List<ModelTypeCountDto> modelTypeCount(String creater);

//	public DetailDto viewById(@Param("modelId") Long modelId);

	public List<ModelCountDto> modelCountByProject(@Param("projectIds") List<Long> projectIds);

	public List<TrainingModelDataDto> queryByCreateTime(@Param("timeString") String timeString);
}
